from django.contrib import admin

from .models import *


@admin.register(Aircraft)
class AircraftAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'first_class_number', 'economy_class')
    list_display_links = ['name', 'first_class_number', 'economy_class']


@admin.register(RouteInfo)
class RouteInfoAdmin(admin.ModelAdmin):
    list_display = ('id','flight_name', 'from_place','to_place','start_time','end_time','first_class_price','economy_class_price')
    list_display_links = ['flight_name']